package com.skh.array;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;

/**
 * @Author: skh
 * @Date: 2020/3/13 10:19
 * @Description: 两个数组的交集 II
 */
public class Intersect {

    /**
     * 给定两个数组，编写一个函数来计算它们的交集。
     *
     * 示例 1:
     *
     * 输入: nums1 = [1,2,2,1], nums2 = [2,1,2]
     * 输出: [2,2]
     * 示例 2:
     *
     * 输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
     * 输出: [4,9]
     * 说明：
     *
     * 输出结果中每个元素出现的次数，应与元素在两个数组中出现的次数一致。
     * 我们可以不考虑输出结果的顺序。
     * 进阶:
     *
     * 如果给定的数组已经排好序呢？你将如何优化你的算法？
     * 如果 nums1 的大小比 nums2 小很多，哪种方法更优？
     * 如果 nums2 的元素存储在磁盘上，磁盘内存是有限的，并且你不能一次加载所有的元素到内存中，你该怎么办？
     */

    /**
     * 思路1:哈希映射
     */
    public int[] intersect(int[] nums1, int[] nums2) {
        /**
         * 我们先在 HashMap 记录一个数组中的存在的数字和对应出现的次数。
         * 然后，我们遍历第二个数组，检查数字在 HashMap 中是否存在，
         * 如果存在且计数为正，则将该数字添加到答案并减少 HashMap 中的计数。
         */

        HashMap<Integer, Integer> map = new HashMap<>();
        for (Integer num : nums1) {
            if (!map.containsKey(num)) {
                map.put(num, 1);
            }else {
                map.put(num, map.get(num) + 1);
            }
        }

        List<Integer> resultList = new ArrayList<>();
        for (int num : nums2) {
            if (map.containsKey(num)) {
                if (map.get(num) > 0) {
                    resultList.add(num);
                    map.put(num, map.get(num) - 1);
                }
            }

        }
        int[] result = resultList.stream().mapToInt(Integer::valueOf).toArray();
        return result;
    }

    /**
     * 思路2:排序
     * 当输入数据是有序的，推荐使用此方法。在这里，我们对两个数组进行排序，并且使用两个指针在一次扫面找出公共的数字。
     * 对数组 nums1 和 nums2 排序。
     * 初始化指针 i，j 和 k 为 0。
     * 指针 i 指向 nums1，指针 j 指向 nums2：
     * 如果 nums1[i] < nums2[j]，则 i++。
     * 如果 nums1[i] > nums2[j]，则 j++。
     * 如果 nums1[i] == nums2[j]，将元素拷贝到 list，且 i++，j++
     * 返回list。
     *
     */
    public int[] intersect1(int[] nums1, int[] nums2) {
        Arrays.sort(nums1);
        Arrays.sort(nums2);
        int i =0;
        int j = 0;

        List<Integer> list = new ArrayList<>();
        while (i < nums1.length && j < nums2.length) {
            if (nums1[i] < nums2[j]) {
                i++;
            } else if (nums1[i] > nums2[j]) {
                j++;
            }else {
                list.add(nums1[i]);
                i++;
                j++;
            }
        }

        int[] ints = list.stream().mapToInt(Integer::valueOf).toArray();
        return ints;

    }

    public static void main(String[] args) {
        Intersect intersect = new Intersect();
        int[] num1 = {1, 2, 2, 1};
        int[] num2 = {2,1, 2};
        int[] intersect1 = intersect.intersect(num1, num2);
        System.out.println(Arrays.toString(intersect1));

    }
}
